package cn.kk.utils;

import java.util.HashMap;
import java.util.Map;

public class ThreadLocalUtil {

    // 使用ThreadLocal存储每个线程的数据
    private static final ThreadLocal<Map<String, Object>> threadLocal = new ThreadLocal<>();

    /**
     * 将数据存入ThreadLocal中
     * @param key   数据的键
     * @param value 数据的值
     */
    public static void put(String key, Object value) {
        Map<String, Object> map = threadLocal.get();
        if (map == null) {
            map = new HashMap<>();
            threadLocal.set(map);
        }
        map.put(key, value);
    }

    /**
     * 获取ThreadLocal中指定键的数据
     * @param key 数据的键
     * @return 数据的值，如果不存在则返回null
     */
    public static Object get(String key) {
        Map<String, Object> map = threadLocal.get();
        if (map != null) {
            return map.get(key);
        }
        return null;
    }

    /**
     * 获取ThreadLocal中的数据
     * @return 数据，如果不存在则返回null
     */
    public static Object get() {
        return threadLocal.get();
    }

    /**
     * 移除ThreadLocal中指定键的数据
     * @param key 数据的键
     */
    public static void remove(String key) {
        Map<String, Object> map = threadLocal.get();
        if (map != null) {
            map.remove(key);
        }
    }

    /**
     * 移除ThreadLocal中当前线程的数据
     */
    public static void remove() {
        threadLocal.remove();
    }

    /**
     * 清除ThreadLocal中所有的数据
     */
    public static void removeAll() {
        threadLocal.remove();
    }
}